package org.finesys.ai.core.embedding;

public enum EmbeddingModel {

    TEXT_EMBEDDING_ADA_002("text-embedding-ada-002"),

    TEXT_EMBEDDING_3_SMALL("text-embedding-3-small"),
    TEXT_EMBEDDING_3_LARGE("text-embedding-3-large"),
    //deepseek
    BGE_M3("bge-m3"),
    NOMIC_EMBED_TEXT("nomic-embed-text"),
    MXBAI_EMBED_LARGE("mxbai-embed-large"),
    SNOWFLAKE_ARCTIC_EMBED("snowflake-arctic-embed"),
    SNOWFLAKE_ARCTIC_EMBED2("snowflake-arctic-embed2"),
    ALL_MINILM("all-minilm"),
    BGE_LARGE("bge-large"),
    PARAPHRASE_MULTILINGUAL("paraphrase-multilingual"),
    GRANITE_EMBEDDING("granite-embedding");

    private final String value;

    EmbeddingModel(String value) {
        this.value = value;
    }

    @Override
    public String toString() {
        return value;
    }
}
